package com.ljy.my_study.leetcode.比特位计数;

import java.util.Arrays;

/**
 * @author lijunying
 * @date 2021/3/3 10:11
 */
public class TestMain {

//    给定一个非负整数 num。对于 0 ≤ i ≤ num 范围中的每个数字 i ，计算其二进制数中的 1 的数目并将它们作为数组返回。
//
//    示例 1:
//
//    输入: 2
//    输出: [0,1,1]
//    示例 2:
//
//    输入: 5
//    输出: [0,1,1,2,1,2]
//    进阶:
//
//    给出时间复杂度为O(n*sizeof(integer))的解答非常容易。但你可以在线性时间O(n)内用一趟扫描做到吗？
//    要求算法的空间复杂度为O(n)。
//    你能进一步完善解法吗？要求在C++或任何其他语言中不使用任何内置函数（如 C++ 中的 __builtin_popcount）来执行此操作。
//
//    来源：力扣（LeetCode）
//    链接：https://leetcode-cn.com/problems/counting-bits
//    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

    public static void main(String[] args) {
        System.out.println(Arrays.toString(new TestMain().countBits(8)));
    }

    public int[] countBits(int num) {
        int[] result=new int[num+1];
        if(num==0){
            return result;
        }
        result[1]=1;
        if(num==1){
            return result;
        }
        int n=1;
        for(int i=2;i<result.length;i++){
            int index=i-n;
            result[i]=result[index];
            if(index!=n){
                result[i]++;
            }else{
                n<<=1;
            }
        }
        return result;
    }
}
